
import os
os.environ['HF_ENDPOINT'] = 'https://hf-mirror.com'
import timm
# m = timm.create_model('resnet18', pretrained=True, pretrained_cfg_overlay=dict(file="C:/Users/ChenmingSong/.cache/huggingface/hub/models--timm--resnet18.a1_in1k/snapshots/491b427b45c94c7fb0e78b5474cc919aff584bbf/model.safetensors"))
# m = timm.create_model('resnet18', pretrained=True)
m = timm.create_model('volo_d1_224', pretrained=True)
m.eval()
print(m)

# 下面这条指令是加载本地的预训练模型，防止网络不好的情况出现，同时放在本地的目录下面。
# model_ft = timm.create_model('resnetrs50', pretrained=True, pretrained_cfg_overlay=dict(file='/home/xiaoxin/Documents/hc/-/bin/pytorch_model.bin'))
